import sys
import os
import argparse

sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from scripts.model_training import retrain_model_with_new_data

# 复训模型
def main():
    # 创建参数解析器
    parser = argparse.ArgumentParser(description='使用新数据复训模型')
    
    # 添加数据路径参数
    parser.add_argument('data_path', help='新数据的路径')
    
    # 解析参数
    args = parser.parse_args()
    
    # 获取数据路径
    new_data_path = args.data_path
    
    # 检查路径是否存在
    if not os.path.exists(new_data_path):
        print(f"错误: 找不到文件 '{new_data_path}'")
        return
    
    # 复训模型
    model = retrain_model_with_new_data(new_data_path)
    
    if model is not None:
        print("模型复训成功！")

if __name__ == '__main__':
    main()